﻿using System;
using System.Collections.Generic;
using System.Linq;
using ServiceStack.Script;
using ServiceStack.Web;

namespace ServiceStack;

public class AutoQueryScripts : ScriptMethods, IAutoQueryDbFilters
{
    private IRequest? req(ScriptScopeContext scope) => scope.GetValue(ScriptConstants.Request) as IRequest;
    private ServiceStackHost appHost => HostContext.AppHost;

    public object sendToAutoQuery(ScriptScopeContext scope, string requestName) => 
        sendToAutoQuery(scope, TypeConstants.EmptyObjectDictionary, requestName, null);
    public object sendToAutoQuery(ScriptScopeContext scope, object dto, string requestName) => sendToAutoQuery(scope, dto, requestName, null);
    public object sendToAutoQuery(ScriptScopeContext scope, object dto, string requestName, object? options)
    {
        try
        {
            if (requestName == null)
                throw new ArgumentNullException(nameof(requestName));
            if (dto == null)
                throw new ArgumentNullException(nameof(dto));

            var requestType = appHost.Metadata.GetOperationType(requestName);
            if (requestType == null)
                throw new ArgumentException("Request DTO not found: " + requestName);

            if (requestType.HasInterface(typeof(IQueryData)))
            {
                if (Context.ScriptMethods.FirstOrDefault(x => x is ServiceStackScripts) is not ServiceStackScripts ssFilter)
                    throw new NotImplementedException(nameof(sendToAutoQuery) + " Data requires " + nameof(ServiceStackScripts));

                return ssFilter.sendToAutoQuery(scope, dto, requestName, options);
            }
                
            var autoQuery = appHost.TryResolve<IAutoQueryDb>();
            if (autoQuery == null)
                throw new NotSupportedException("The AutoQueryFeature plugin is not registered.");

            var objDictionary = dto is Dictionary<string, object> od ? od : null;
                
            var requestDto = objDictionary != null 
                ? objDictionary.FromObjectDictionary(requestType)
                : dto.GetType() == requestType
                    ? dto
                    : dto.ConvertTo(requestType);
                
            if (requestDto is not IQueryDb aqDto)
                throw new ArgumentException("Request DTO is not an AutoQuery DTO: " + requestName);

            var reqParams = objDictionary?.ToStringDictionary() ?? TypeConstants.EmptyStringDictionary;
            var requestDtoType = requestDto.GetType();
            var fromType = autoQuery.GetFromType(requestDtoType);
            var httpReq = req(scope);
            using var db = autoQuery.GetDb(fromType, httpReq);
            var q = autoQuery.CreateQuery(aqDto, reqParams, httpReq, db);
            var response = autoQuery.Execute(aqDto, q, db);
            return response;
        }
        catch (Exception ex)
        {
            throw new StopFilterExecutionException(scope, options, ex);
        }
    }
}